package com.thp.bigdata;

import java.io.IOException;
import java.io.UnsupportedEncodingException;
import java.util.List;

import org.apache.zookeeper.CreateMode;
import org.apache.zookeeper.KeeperException;
import org.apache.zookeeper.WatchedEvent;
import org.apache.zookeeper.Watcher;
import org.apache.zookeeper.ZooDefs.Ids;
import org.apache.zookeeper.ZooKeeper;
import org.apache.zookeeper.data.Stat;
import org.junit.Before;
import org.junit.Test;

/**
 * zookeeper的客户端 数据的增删改查,已经对数据的监听跟响应
 * 
 * @author tommy
 *
 */
public class SimpleZkClient {

	// zookeeper集群里面只要有半数存活就可用
	
	/**
	 * 注意点:  如果是直接使用的 ip地址的话,那么我测试的时候总会报错,ConnectionLoss 的错误
	 *         在host文件里面配置了IP地址 然后使用主机名的话,程序是可以运行的
	 */
	
	// private static final String connectString = "192.168.17.150:2181,192.168.17.151:2181,192.168.17.153:2181";
	
	private static final String connectString = "bd1:2181,bd2:2181,bd3:2181";

	// 超时时间
	private static final int sessionTimeout = 2000;

	ZooKeeper zkClient = null;

	@Before
	public void init() throws IOException {
		zkClient = new ZooKeeper(connectString, sessionTimeout, new Watcher() {
			// zookeeper可以根据事件作出相应的响应
			@Override
			public void process(WatchedEvent event) {
				// 收到事件通知后的回调函数 (应该是我们自己的事件处理逻辑)
				System.out.println(event.getType() + " -- " + event.getPath()); // 事件的类型  --   事件发生的节点
				try {
					// getChildren 只是为了 设置这个监听  相当于又监听了
					zkClient.getChildren("/", true);
				} catch (KeeperException | InterruptedException e) {
					e.printStackTrace();
				}
			}
		});
	}

	/**
	 * 数据的增删改查
	 * 
	 * @throws InterruptedException
	 * @throws KeeperException
	 */

	// 创建数据节点到zookeeper中
	@Test
	public void testCreate() throws KeeperException, InterruptedException {

		// 第一个参数 要创建节点的路径
		// 第二个参数 data 数据可以是任意的 只需要转换成byte类型的数组就可以
		// 第三个参数是 权限: Ids.CREATOR_ALL_ACL --- 是一个枚举 可以选择对应的权限 一般选择
		// OPEN_ACL_UNSAFE
		// 第四个参数 节点的类型 四种可供选择
		String nodeCreated = zkClient.create("/eclipse3", "hellozk".getBytes(), Ids.OPEN_ACL_UNSAFE,
				CreateMode.PERSISTENT);

	}
	
	// 获取子节点
	@Test
	public void getChildren() throws KeeperException, InterruptedException {
		List<String> children = zkClient.getChildren("/", true);
		for(String child : children) {
			System.out.println(child);
		}
		
		// 睡一段时间 可以查看到监听的变化    在zookeeper客户端 进行修改
		// 注意 :  这个监听器只监听一次
		Thread.sleep(Long.MAX_VALUE);   // NodeChildrenChanged -- /
		
	}
	
	
	
	// 判断znode是否存在
	@Test
	public void testExist() throws KeeperException, InterruptedException {
		Stat stat = zkClient.exists("/eclipse1", false);
		System.out.println(stat == null ? "not exist" : "exist");
	}
	
	
	// 获取znode数据
	@Test
	public void getData() throws KeeperException, InterruptedException, UnsupportedEncodingException {
		byte[] data = zkClient.getData("/eclipse1", false, null);
		System.out.println(new String(data, "utf-8"));
	}
	
	// 删除znode数据
	@Test
	public void deleteZnode() throws KeeperException, InterruptedException, UnsupportedEncodingException {
		// 参数2是指定删除的版本  -1 表示删除所有的版本
		zkClient.delete("/eclipse1", -1);
	}
	
	// 修改znode数据
	@Test
	public void setData() throws KeeperException, InterruptedException, UnsupportedEncodingException {
		zkClient.setData("/app1", "i lost you".getBytes(), -1);
		byte[] data = zkClient.getData("/app1", false, null);
		System.out.println(new String(data));
	}
	
	
	
	

}
